from base import *
from tag import Tag
from django.utils.translation import ugettext as _

class RootNodeManager(NodeManager):
    def search(self, keywords):
        return False, self.filter(models.Q(title__icontains=keywords) | models.Q(body__icontains=keywords))

class RootNode(Node):
    """
    This is an "abstract" class for the primary nodes, such as Question, 
    Issue, or Argument, for which can take "replies" such as Answer,
    ProCon, or Rebuttal.
    """
    class Meta(Node.Meta):
        proxy = True

    answer_count = DenormalizedField("children", ~models.Q(state_string__contains="(deleted)"), node_type="answer")
    accepted_count = DenormalizedField("children", ~models.Q(state_string__contains="(deleted)"), node_type="answer", marked=True)
    favorite_count = DenormalizedField("actions", action_type="favorite", canceled=False)

    objects = RootNodeManager()

    @property
    def closed(self):
        return self.nis.closed

    @property    
    def view_count(self):
        return self.extra_count

    @property
    def headline(self):
        return self._headline()

    def _headline(self):
        if self.nis.deleted:
            return _('[deleted] ') + self.title

        if self.nis.closed:
            return _('[closed] ') + self.title

        if not self.nis.published:
            return _('[draft] ') + self.title

        return self._other_headlines()

    def _other_headlines(self): # allow specialized headlines, like for promoted
        return self.title

    @models.permalink    
    def get_absolute_url(self):
        return ('issue', (), {'id': self.id, 'slug': django_urlquote(slugify(self.title))})
        
    def meta_description(self):
        pass

    def get_revision_url(self):
        pass

    def get_related(self, count=10):
        cache_key = '%s.related:%d:%d' % (settings.APP_URL, count, self.id)
        related_list = cache.get(cache_key)

        if related_list is None:
            related_list = RootNode.objects.filter_state(deleted=False,published=True).values('id').filter(tags__id__in=[t.id for t in self.tags.all()]
            ).exclude(id=self.id).annotate(frequency=models.Count('id')).order_by('-frequency')[:count]
            cache.set(cache_key, related_list, 60 * 60)

        return [RootNode.objects.get(id=r['id']) for r in related_list]



class RootNodeSubscription(models.Model):
    user = models.ForeignKey(User)
    issue = models.ForeignKey(Node)
    auto_subscription = models.BooleanField(default=True)
    last_view = models.DateTimeField(default=datetime.datetime.now())

    class Meta:
        app_label = 'forum'


class RootNodeRevision(NodeRevision):
    class Meta:
        proxy = True

